#pragma once
#include <jsoncpp/json/json.h>
#include <signal.h>
#include "compiler.hpp"
#include "runner.hpp"
#include "../comm/Log.hpp"
#include "../comm/Util.hpp"

namespace compile_and_run_ns
{
    using namespace log_ns;
    using namespace util_ns;
    using namespace compiler_ns;
    using namespace runner_ns;

    class Compile_Run
    {
    public:
        static void RemoveTempFile(const std::string &file_name)
        {
            std::string _src = PathUtil::Src(file_name);
            if(FileUtil::IsExistfile(_src)) unlink(_src.c_str());

            std::string _execute = PathUtil::Exe(file_name);
            if(FileUtil::IsExistfile(_execute)) unlink(_execute.c_str());

            std::string _compile_error = PathUtil::CompileError(file_name);
            if(FileUtil::IsExistfile(_compile_error)) unlink(_compile_error.c_str());

            std::string _stdin = PathUtil::Stdin(file_name);
            if(FileUtil::IsExistfile(_stdin)) unlink(_stdin.c_str());

            std::string _stdout = PathUtil::Stdout(file_name);
            if(FileUtil::IsExistfile(_stdout)) unlink(_stdout.c_str());

            std::string _stderr = PathUtil::Stderr(file_name);
            if(FileUtil::IsExistfile(_stderr)) unlink(_stderr.c_str());
        }

        // >0 可能还有其他信号
        static std::string CodeToDesc(int code)
        {
            std::string desc;
            if(code == SIGABRT)//6
            {
                desc = "内存超出范围";
            }
            else if(code == SIGXCPU)//24
            {
                desc = "CPU超时";
            }
            else if(code == SIGFPE)//8
            {
                desc = "浮点数溢出错误";
            }
            else 
            {
                desc = "未知: " + std::to_string(code);
            }
            return desc;
        }

        // 输入:
        // code  用户提交的代码
        // input 用户提交的代码对应的输入
        // cpu_limit  时间要求
        // mem_limit  空间要求

        // 输出:
        // status  状态码
        // reason  状态码对应的信息
        // stdout  程序的结果信息
        // stderr  程序的错误信息
        static void Start(const std::string &in_json, std::string *out_json)
        {
            // 反序列化
            Json::Value in_value;
            Json::Reader reader;
            reader.parse(in_json, in_value);
            std::string code = in_value["code"].asString();
            std::string input = in_value["input"].asString();
            int cpu_limit = in_value["cpu_limit"].asInt();
            int mem_limit = in_value["mem_limit"].asInt();

            // 序列化
            Json::Value out_value;

            if(code.size() == 0)
            {
                out_value["status"] = -1;
                out_value["reason"] = "代码为空";
                // 序列化过程
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }

            // 形成的文件具有唯一性，没有目录和后缀
            // 毫秒级时间戳+原子性递增唯一值：来保证唯一性
            std::string file_name = FileUtil::UniqFileName();

            // 形成临时src文件
            if(!FileUtil::WriteFile(PathUtil::Src(file_name), code))
            {
                out_value["status"] = -2;
                out_value["reason"] = "代码有错误";
                // 序列化过程
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }

            // 编译
            if(!Compiler::Compile(file_name))
            {
                out_value["status"] = -3;
                std::string _compile_error;
                FileUtil::ReadFile(PathUtil::CompileError(file_name), &_compile_error, true);
                out_value["reason"] = _compile_error;
                // 序列化过程
                Json::StyledWriter writer;
                *out_json = writer.write(out_value);
                return;
            }
            
            // 运行
            int status_code = Runner::Run(file_name, cpu_limit, mem_limit);
            if(status_code < 0)
            {
                out_value["status"] = -2;
                out_value["reason"] = "运行有错误"; 
            }
            else if(status_code > 0)
            {
                out_value["status"] = status_code; // 信号(报错)
                out_value["reason"] = CodeToDesc(status_code); // 信号描述错误
            }
            else 
            {
                // 运行成功 
                out_value["status"] = 0;
                out_value["reason"] = "编译并运行成功";
                std::string _stdout;
                FileUtil::ReadFile(PathUtil::Stdout(file_name), &_stdout, true);
                out_value["stdout"] = _stdout;

                std::string _stderr;
                FileUtil::ReadFile(PathUtil::Stderr(file_name), &_stderr, true);
                out_value["stderr"] = _stderr;
            }
            // 序列化过程
            Json::StyledWriter writer;
            *out_json = writer.write(out_value);
 
            // 清理临时文件 
            RemoveTempFile(file_name); 
        }
    };
}